package com.springai.repository;

import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Component
 @RequiredArgsConstructor
 public class InMemoryChatHistoryRepository implements ChatHistoryRepository {
 private final Map<String, List<String>> chatHistory = new HashMap<>();

 @Override
 public void save(String type, String chatId) {
 /*if (!chatHistory.containsKey(type)) {
 chatHistory.put(type, new ArrayList<>());
 }
 List<String> chatIds = chatHistory.get(type);*/
  List<String> chatIds = chatHistory.computeIfAbsent(type, k -> new
          ArrayList<>());
  if (chatIds.contains(chatId)) {
   return;
  }
  chatIds.add(chatId);
 }
 @Override
 public List<String> getChatIds(String type) {
 /*List<String> chatIds = chatHistory.get(type);
 return chatIds == null ? List.of() : chatIds;*/
  return chatHistory.getOrDefault(type, List.of());
 }
}
